On-Policy Distillation with Negative-Policy Rollouts
基于负策略轨迹的在线蒸馏
🧭 LLM 训练 · 在线蒸馏负样本 ★ 深度解读
NAVER AI Lab
在 OPD 的 rollout 阶段混入低能力负策略轨迹,不改变奖励公式即可注入显式负信号,显著提升推理性能。
📌 工作参考价值(4 点)
- 在 LLM 微调/蒸馏中引入弱模型负样本 rollout:如果业务中有同模型系的低能力版本(如小参数量的 base 或旧版模型),可以将其作为负策略,按一定比例混入 on-policy 训练 batch,不动 reward 即可让模型远离低质量输出,相当于给 OPD 加了一个“反向参考”。
- 负策略 rollout 可以预生成并缓存复用:论文显示纯负策略 rollout(α=1)时,训练时间从 9h19m 降到 3h34m(排除预生成),这对落地 LLM 蒸馏训练的成本优化很实用。
- 负样本选择有讲究:不是随便用差模型,而是同模型家族、能力明显低于 student 的版本;简单加大 temperature、persona prompt 或 drop layer 等降级 student 的方法也能带来少量提升,但远不如单独负策略(+3.4 vs +0.9~1.0 avg)。
- 可复用分析指标 NSP/NOR 做训练调试:NSP 衡量抑制是否集中在负策略偏好 token 上,NOR 衡量学生与负策略的重叠是否下降;二者与最终性能正相关,可用于监控负信号是否按预期生效。